package com.songjia.zuul.filter;

import javax.servlet.http.HttpServletRequest;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;

/**
 * 
 * 
 * @Description: ZUUL请求过滤。继承抽象类ZuulFilter，重写ZuulFilter的四个抽象方法即可完成对请求的拦截和过滤。
 *
 * @author: songjia 
 * @date: 2018年3月22日 上午10:13:49
 */
public class AccessFilter extends ZuulFilter {

	private static final Logger LOG = LoggerFactory.getLogger(AccessFilter.class);

	/**
	 * filterType：返回一个字符串代表过滤器的类型，在zuul中定义了四种不同生命周期的过滤器类型，具体如下：
	 * 
	 * pre：路由之前
	 * 
	 * routing：路由之时
	 * 
	 * post： 路由之后
	 * 
	 * error：发送错误调用
	 * 
	 * @return
	 * @see com.netflix.zuul.ZuulFilter#filterType()
	 */
	@Override
	public String filterType() {
		return "pre";
	}

	/**
	 * filterOrder：过滤器的执行顺序，当请求在一个阶段中存在多个过滤器时，需要根据该方法返回的值来依次执行
	 * 
	 * @return
	 * @see com.netflix.zuul.ZuulFilter#filterOrder()
	 */
	@Override
	public int filterOrder() {
		return 0;
	}

	/**
	 * shouldFilter：判断该过滤器是否需要被执行。返回true表示执行，返回false表示不执行。可以利用此方法来指定过滤器的执行范围
	 * 
	 * @return
	 * @see com.netflix.zuul.IZuulFilter#shouldFilter()
	 */
	@Override
	public boolean shouldFilter() {
		return true;
	}

	/**
	 * run：过滤器的具体逻辑。可用很复杂，包括查sql，nosql去判断该请求到底有没有权限访问。
	 * 
	 * @return
	 * @see com.netflix.zuul.IZuulFilter#run()
	 */
	@Override
	public Object run() {
		RequestContext ctx = RequestContext.getCurrentContext();
		HttpServletRequest request = ctx.getRequest();
		LOG.info(String.format("%s >>> %s", request.getMethod(), request.getRequestURL().toString()));
		Object accessToken = request.getParameter("token");
		if (accessToken == null) {
			LOG.warn("token is empty");
			// 令zuul过滤当前请求，不进行路由
			ctx.setSendZuulResponse(false);
			// 设置错误码病返回
			ctx.setResponseStatusCode(401);
			// 编辑返回的body内容
			ctx.setResponseBody("token is empty");
			return null;
		}
		LOG.info("access token ok");
		return null;
	}
}
